package com.xyy.saas.payment.util.jdpay;

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.FileUtils;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Enumeration;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;

@Slf4j
public class FileUtil {

    /**
     * 删除文件
     */
    public static void deleteFile(String logPrefix, String filePath) {
        File file = new File(filePath);
        if (file.exists()) {
            log.info("{} 删除文件：{}", logPrefix, filePath);
            file.delete();
        }
    }

    /**
     * 写入数据到文件
     */
    public static void writeFile(String filePath, String fileName, String data, boolean append) throws IOException {
        File file = new File(filePath);
        if (!file.exists() && !file.isDirectory()) {
            file.mkdirs();
        }
        FileUtils.writeStringToFile(new File(filePath + File.separator + fileName), data, "UTF-8", append);
    }

    /**
     * 对读到的账单数据做转换
     */
    private static String transCol(String msg){
        if (StringUtils.isEmpty(msg) || !msg.startsWith("=\"")) {
            return msg;
        }
        if (StringUtils.isEmpty(msg) || msg.length() <= 3) {
            return "";
        }
        return msg.substring(2, msg.length() - 1);
    }

    /**
     * zip解压
     * @param srcZipFile      zip源文件
     * @param destDirPath     解压后的目标文件夹
     */
    public static void unZip(String logPrefix, String srcZipFile, String destDirPath) throws Exception {
        File srcFile = new File(srcZipFile);
        // 判断源文件是否存在
        if (!srcFile.exists() || srcFile.length() == 0) {
            log.info("{} [jdPay]账单处理: 解压文件[{}]不存在，不进行账单解压", logPrefix, srcZipFile);
            return;
        }
        // 开始解压
        ZipFile zipFile = null;
        try {
            zipFile = new ZipFile(srcFile);
            Enumeration<?> entries = zipFile.entries();
            while (entries.hasMoreElements()) {
                ZipEntry entry = (ZipEntry) entries.nextElement();
                // 如果是文件夹，就创建个文件夹
                if (entry.isDirectory()) {
                    String dirPath = destDirPath + "/" + entry.getName();
                    File dir = new File(dirPath);
                    dir.mkdirs();
                } else {
                    InputStream is = null;
                    FileOutputStream fos = null;
                    try {
                        // 如果是文件，就先创建一个文件，然后用io流把内容copy过去
                        File targetFile = new File(destDirPath + "/" + entry.getName());
                        // 保证这个文件的父文件夹必须要存在
                        if (!targetFile.getParentFile().exists()) {
                            targetFile.getParentFile().mkdirs();
                        }
                        targetFile.createNewFile();
                        // 将压缩文件内容写入到这个文件中
                        is = zipFile.getInputStream(entry);
                        fos = new FileOutputStream(targetFile);
                        int len;
                        byte[] buf = new byte[1024];
                        while ((len = is.read(buf)) != -1) {
                            fos.write(buf, 0, len);
                        }
                    } catch (Exception e) {
                        throw e;
                    } finally {
                        // 关流顺序，先打开的后关闭
                        if (null != fos) {
                            fos.close();
                        }
                        if (null != is) {
                            is.close();
                        }
                    }
                }
            }
        } catch (Exception e) {
            throw new Exception("[jdPay]账单处理: unzip error from ZipUtils", e);
        } finally {
            if(zipFile != null){
                try {
                    zipFile.close();
                } catch (IOException e) {
                }
            }
        }
    }

}
